import torchaudio
from config import HP


if __name__ == '__main__':
    '''
    SpeechCommand v2下载：
    将url参数改为 speech_commands_v0.02 即可
    '''
    speech_command_dataset = torchaudio.datasets.SPEECHCOMMANDS(
        HP.data_root,
        url='speech_commands_v0.01',
        download=True
    )

    print('=========================')
    print('Dataset Download Success ')
    print('=========================')
